package lccup;

/**
 * https://leetcode-cn.com/contest/season/2020-fall/problems/2vYnGI/
 * https://leetcode-cn.com/problems/2vYnGI/
 * 2. 早餐组合
 * 通过的用户数2027
 * 尝试过的用户数3192
 * 用户总通过次数2066
 * 用户总提交次数12437
 * 题目难度Easy
 * 小扣在秋日市集选择了一家早餐摊位，一维整型数组 staple 中记录了每种主食的价格，一维整型数组 drinks 中记录了每种饮料的价格。小扣的计划选择一份主食和一款饮料，且花费不超过 x 元。请返回小扣共有多少种购买方案。
 * <p>
 * 注意：答案需要以 1e9 + 7 (1000000007) 为底取模，如：计算初始结果为：1000000008，请返回 1
 * <p>
 * 示例 1：
 * <p>
 * 输入：staple = [10,20,5], drinks = [5,5,2], x = 15
 * <p>
 * 输出：6
 * <p>
 * 解释：小扣有 6 种购买方案，所选主食与所选饮料在数组中对应的下标分别是：
 * 第 1 种方案：staple[0] + drinks[0] = 10 + 5 = 15；
 * 第 2 种方案：staple[0] + drinks[1] = 10 + 5 = 15；
 * 第 3 种方案：staple[0] + drinks[2] = 10 + 2 = 12；
 * 第 4 种方案：staple[2] + drinks[0] = 5 + 5 = 10；
 * 第 5 种方案：staple[2] + drinks[1] = 5 + 5 = 10；
 * 第 6 种方案：staple[2] + drinks[2] = 5 + 2 = 7。
 * <p>
 * 示例 2：
 * <p>
 * 输入：staple = [2,1,1], drinks = [8,9,5,1], x = 9
 * <p>
 * 输出：8
 * <p>
 * 解释：小扣有 8 种购买方案，所选主食与所选饮料在数组中对应的下标分别是：
 * 第 1 种方案：staple[0] + drinks[2] = 2 + 5 = 7；
 * 第 2 种方案：staple[0] + drinks[3] = 2 + 1 = 3；
 * 第 3 种方案：staple[1] + drinks[0] = 1 + 8 = 9；
 * 第 4 种方案：staple[1] + drinks[2] = 1 + 5 = 6；
 * 第 5 种方案：staple[1] + drinks[3] = 1 + 1 = 2；
 * 第 6 种方案：staple[2] + drinks[0] = 1 + 8 = 9；
 * 第 7 种方案：staple[2] + drinks[2] = 1 + 5 = 6；
 * 第 8 种方案：staple[2] + drinks[3] = 1 + 1 = 2；
 * <p>
 * 提示：
 * <p>
 * 1 <= staple.length <= 10^5
 * 1 <= drinks.length <= 10^5
 * 1 <= staple[i],drinks[i] <= 10^5
 * 1 <= x <= 2*10^5
 */

import java.util.Arrays;

/**
 * 题解两种思路：
 * 1、双循环的方式，计算出总的购买方案。 时间复杂度:O（N^2）;空间复杂度：O(1)
 * 2、先排序，再进行循环，排序后可以避免明显不可能的组合方案，减少无意义的计算。时间复杂度:O（M^2+2*Nlog(N)）;空间复杂度：O(1)
 */
public class Lccup_20_breakfast_combination {

    public static void main(String[] args) {
        int[] staple = {10, 20, 5};
        int[] drinks = {5, 5, 2};
        System.out.println(breakfastNumber(staple, drinks, 15));

    }

    /**
     * 解法3： 双指针法
     * 时间复杂度： O(M+N)
     * 空间复杂度： O(1)
     *
     * @param staple
     * @param drinks
     * @param x
     * @return
     */
    public static int breakfastNumber(int[] staple, int[] drinks, int x) {
        Arrays.sort(staple);
        Arrays.sort(drinks);
        int count = 0;
        int i = 0;
        int m = staple.length - 1;
        int n = drinks.length - 1;
        while (i <= m && n >= 0) {
            if (staple[i] + drinks[n] <= x) {
                count = (count+n+1)%1000000007;
                i++;
            }else{
                n--;
            }
        }
        return count%1000000007;

    }


    /**
     * 题解1:
     * 双循环的方式；排序法
     *
     * @param staple
     * @param drinks
     * @param x
     * @return
     */
    public static int breakfastNumber2(int[] staple, int[] drinks, int x) {
        Arrays.sort(staple);
        Arrays.sort(drinks);
        int count = 0;
        for (int i : staple) {
            if (i >= x) {
                break;
            }
            for (int j : drinks) {
                if (i + j <= x) {
                    count++;
                    if (count >= 1000000007) {
                        count %= 1000000007;
                    }
                } else {
                    break;
                }
            }
        }
        return count;

    }

    /**
     * 题解1:
     * 双循环的方式
     * 时间复杂度 ： O(N^2)
     * 空间复杂度 ： O(N^2)
     *
     * @param staple
     * @param drinks
     * @param x
     * @return
     */
    public static int breakfastNumber1(int[] staple, int[] drinks, int x) {
        int count = 0;
        for (int i : staple) {
            if (i >= x) {
                continue;
            }
            for (int j : drinks) {
                if (i + j <= x) {
                    count++;
                }
            }
        }
        return count;
    }


}
